// 从 @langchain/community 包中引入智谱AI的聊天模型
import { ChatZhipuAI } from "@langchain/community/chat_models/zhipuai";
// （可选）引入 OpenAI 聊天模型，当前被注释掉
// import * as ChatOpenAI from "@langchain/openai";

// 导出一个用于创建AI模型实例的函数
export const createAIModel = () => {
  // 可选：根据环境变量选择模型类型，当前默认使用智谱（已注释）
  // const modelType = process.env.AI_MODEL_TYPE || "zhipu"; // 默认使用智普

  // 返回一个 ChatZhipuAI 实例，配置相关参数
  return new ChatZhipuAI({
    apiKey: process.env.ZHIPU_API_KEY, // 从环境变量中读取智谱API密钥
    modelName: "glm-4-flash", // 指定使用的模型名称（可选 "glm-3-turbo"）
    temperature: 0.7, // 设置生成内容的多样性，数值越高内容越随机
    streaming: true, // 启用流式输出，提升响应速度
  });
};
